from config import Config
import code2vec

if __name__ == '__main__':
    train_data = r"F:\博士\看的论文\data\code2vec\java-small\java-small"
    test_data = r"F:\博士\看的论文\data\code2vec\java-small\java-small.test.c2v"
    model_dir = r"./models/java-small"
    # release_data=r"I:\博士\看的论文\code2vecdata\java14m_model\models\java14_model\saved_model_iter8.release"
    # release_data = r"E:\论文汇报\10\data\models\java14_model\saved_model_iter8.release"
    release_data=r"F:\博士\看的论文\data\java14_model\saved_model_iter8.release"
    config = Config()
    config.DL_FRAMEWORK = 'tensorflow'
    # config.args = config.arguments_parser().parse_args(["--data", train_data,
    #                                                     "--test", test_data,
    #                                                     "--save", model_dir
    #                                                    ])
    config.args = config.arguments_parser().parse_args(["--load", release_data,
                                                        "--predict"])

    config.setParameter(set_defaults=True, load_from_args=True, verify=True)

    code2vec.run(config)
